# 使用官方 Python 镜像作为基础镜像
FROM python:3.12-slim

# 设置工作目录
WORKDIR /python/flaskapp_ver0.2

# 将当前目录下的所有文件复制到容器的 /python/flaskapp_ver0.2/ 目录
COPY .. /python/flaskapp_ver0.2/

# 安装 requirements.txt 中的依赖项
RUN pip install --no-cache-dir -i https://mirrors.aliyun.com/pypi/simple/ -r requirements.txt \
    && pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu \
    && pip install --no-cache-dir -i https://mirrors.aliyun.com/pypi/simple/ pytorch-lightning \
    && rm -rf /root/.cache/pip


CMD ["python", "app.py"]